﻿// ConsoleTools
// Copyright (C) 2017-2024 Dust in the Wind
// 
// This program is free software: you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
// 
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU General Public License for more details.
// 
// You should have received a copy of the GNU General Public License
// along with this program.  If not, see <http://www.gnu.org/licenses/>.

// --------------------------------------------------------------------------------
// Bugs or feature requests
// --------------------------------------------------------------------------------
// Note: For any bug or feature request please add a new issue on GitHub: https://github.com/lastunicorn/ConsoleTools/issues/new/choose

using System;

namespace DustInTheWind.ConsoleTools;

public partial class CustomConsole
{
    /// <summary>
    /// Writes the specified text to the Console using the specified foreground and background colors.
    /// </summary>
    /// <param name="foregroundColor">The foreground color used to write the text.</param>
    /// <param name="backgroundColor">The background color used to write the text.</param>
    /// <param name="text">The text to be written to the Console.</param>
    public static void Write(ConsoleColor foregroundColor, ConsoleColor backgroundColor, string text)
    {
        ConsoleColor initialForegroundColor = Console.ForegroundColor;
        ConsoleColor initialBackgroundColor = Console.BackgroundColor;

        Console.ForegroundColor = foregroundColor;
        Console.BackgroundColor = backgroundColor;

        Console.Write(text);

        Console.ForegroundColor = initialForegroundColor;
        Console.BackgroundColor = initialBackgroundColor;
    }

    /// <summary>
    /// Writes the text representation of the specified array of objects to the Console
    /// using the specified format information and foreground and background colors.
    /// </summary>
    /// <param name="foregroundColor">The foreground color used to write the text.</param>
    /// <param name="backgroundColor">The background color used to write the text.</param>
    /// <param name="format">A composite format string.</param>
    /// <param name="arg">An array of objects to write using format.</param>
    public static void Write(ConsoleColor foregroundColor, ConsoleColor backgroundColor, string format, params object[] arg)
    {
        ConsoleColor initialForegroundColor = Console.ForegroundColor;
        ConsoleColor initialBackgroundColor = Console.BackgroundColor;

        Console.ForegroundColor = foregroundColor;
        Console.BackgroundColor = backgroundColor;

        Console.Write(format, arg);

        Console.ForegroundColor = initialForegroundColor;
        Console.BackgroundColor = initialBackgroundColor;
    }

    /// <summary>
    /// Writes the text representation of the specified object to the Console
    /// using the specified foreground and background colors.
    /// </summary>
    /// <param name="foregroundColor">The foreground color used to write the text.</param>
    /// <param name="backgroundColor">The background color used to write the text.</param>
    /// <param name="o">The value to write.</param>
    public static void Write(ConsoleColor foregroundColor, ConsoleColor backgroundColor, object o)
    {
        ConsoleColor initialForegroundColor = Console.ForegroundColor;
        ConsoleColor initialBackgroundColor = Console.BackgroundColor;

        Console.ForegroundColor = foregroundColor;
        Console.BackgroundColor = backgroundColor;

        Console.Write(o);

        Console.ForegroundColor = initialForegroundColor;
        Console.BackgroundColor = initialBackgroundColor;
    }

    /// <summary>
    /// Writes the specified string value, followed by the current line terminator, to the Console
    /// using the specified foreground and background colors.
    /// </summary>
    /// <param name="foregroundColor">The foreground color used to write the text.</param>
    /// <param name="backgroundColor">The background color used to write the text.</param>
    /// <param name="text">The text to write.</param>
    public static void WriteLine(ConsoleColor foregroundColor, ConsoleColor backgroundColor, string text)
    {
        ConsoleColor initialForegroundColor = Console.ForegroundColor;
        ConsoleColor initialBackgroundColor = Console.BackgroundColor;

        Console.ForegroundColor = foregroundColor;
        Console.BackgroundColor = backgroundColor;

        Console.WriteLine(text);

        Console.ForegroundColor = initialForegroundColor;
        Console.BackgroundColor = initialBackgroundColor;
    }

    /// <summary>
    /// Writes the text representation of the specified array of objects,
    /// followed by the current line terminator, to the Console
    /// using the specified format information, foreground and background colors.
    /// </summary>
    /// <param name="foregroundColor">The foreground color used to write the text.</param>
    /// <param name="backgroundColor">The background color used to write the text.</param>
    /// <param name="format">A composite format string.</param>
    /// <param name="arg">An array of objects to write using format.</param>
    public static void WriteLine(ConsoleColor foregroundColor, ConsoleColor backgroundColor, string format, params object[] arg)
    {
        ConsoleColor initialForegroundColor = Console.ForegroundColor;
        ConsoleColor initialBackgroundColor = Console.BackgroundColor;

        Console.ForegroundColor = foregroundColor;
        Console.BackgroundColor = backgroundColor;

        Console.WriteLine(format, arg);

        Console.ForegroundColor = initialForegroundColor;
        Console.BackgroundColor = initialBackgroundColor;
    }

    /// <summary>
    /// Writes the text representation of the specified object, followed by the current line terminator, to the Console
    /// using the specified foreground and background colors.
    /// </summary>
    /// <param name="foregroundColor">The foreground color used to write the text.</param>
    /// <param name="backgroundColor">The background color used to write the text.</param>
    /// <param name="o">The value to write.</param>
    public static void WriteLine(ConsoleColor foregroundColor, ConsoleColor backgroundColor, object o)
    {
        ConsoleColor initialForegroundColor = Console.ForegroundColor;
        ConsoleColor initialBackgroundColor = Console.BackgroundColor;

        Console.ForegroundColor = foregroundColor;
        Console.BackgroundColor = backgroundColor;

        Console.WriteLine(o);

        Console.ForegroundColor = initialForegroundColor;
        Console.BackgroundColor = initialBackgroundColor;
    }

    /// <summary>
    /// Executes the specified action while the foreground and background colors are set to the specified values.
    /// </summary>
    public static void WithColors(ConsoleColor? foregroundColor, ConsoleColor? backgroundColor, Action action)
    {
        if (action == null) throw new ArgumentNullException(nameof(action));

        if (foregroundColor.HasValue)
        {
            if (backgroundColor.HasValue)
                WithColors(foregroundColor.Value, backgroundColor.Value, action);
            else
                WithForegroundColor(foregroundColor.Value, action);
        }
        else
        {
            if (backgroundColor.HasValue)
                WithBackgroundColor(backgroundColor.Value, action);
            else
                action();
        }
    }

    /// <summary>
    /// Executes the specified action while the foreground and background colors
    /// are set to the specified values.
    /// </summary>
    public static void WithColors(ConsoleColor foregroundColor, ConsoleColor backgroundColor, Action action)
    {
        if (action == null) throw new ArgumentNullException(nameof(action));

        ConsoleColor initialForegroundColor = Console.ForegroundColor;
        ConsoleColor initialBackgroundColor = Console.BackgroundColor;

        Console.ForegroundColor = foregroundColor;
        Console.BackgroundColor = backgroundColor;

        try
        {
            action();
        }
        finally
        {
            Console.ForegroundColor = initialForegroundColor;
            Console.BackgroundColor = initialBackgroundColor;
        }
    }

    /// <summary>
    /// Executes the specified function while the foreground and background colors
    /// are set to the specified values.
    /// </summary>
    public static T WithColors<T>(ConsoleColor? foregroundColor, ConsoleColor? backgroundColor, Func<T> func)
    {
        if (func == null) throw new ArgumentNullException(nameof(func));

        if (foregroundColor.HasValue)
        {
            return backgroundColor.HasValue
                ? WithColors(foregroundColor.Value, backgroundColor.Value, func)
                : WithForegroundColor(foregroundColor.Value, func);
        }

        return backgroundColor.HasValue
            ? WithBackgroundColor(backgroundColor.Value, func)
            : func();
    }

    /// <summary>
    /// Executes the specified function while the foreground and background colors
    /// are set to the specified values.
    /// </summary>
    public static T WithColors<T>(ConsoleColor foregroundColor, ConsoleColor backgroundColor, Func<T> func)
    {
        if (func == null) throw new ArgumentNullException(nameof(func));

        ConsoleColor initialForegroundColor = Console.ForegroundColor;
        ConsoleColor initialBackgroundColor = Console.BackgroundColor;

        Console.ForegroundColor = foregroundColor;
        Console.BackgroundColor = backgroundColor;

        try
        {
            return func();
        }
        finally
        {
            Console.ForegroundColor = initialForegroundColor;
            Console.BackgroundColor = initialBackgroundColor;
        }
    }
}